# 按间距中的绿色按钮以运行脚本。

from setting import *
from whisper import trans2


def chat(prompt="你是一个医生,正在生成病历", model="text-davinci-003"):
    completion = openai.Completion.create(model=model,
                                              prompt=prompt,
                                              max_tokens=1024,
                                              top_p=1,
                                              stop=['STOP'],
                                              temperature=0.8)

    message = completion.choices[0].text
    print(message)


def chat35(prompt="你是一个医生，正在为病人记录病症"):
    completion = openai.ChatCompletion.create(model="gpt-3.5-turbo",
                                              messages=[{"role": "user", "content": prompt}])

    message = completion.choices[0].message
    print(message.content)


if __name__ == '__main__':
    init()
    chat(prompt="您是一个医生专家，请你帮患者生成病历。\n患者基本信息：年龄:35岁、性别:男、姓名:张三。症状：头疼、眼花、失眠。请帮患者生成一份病历。")
    print("=========================")

    chat35(prompt="您是一个医生专家，请你帮患者生成病历。\n患者基本信息：年龄:35岁、性别:男、姓名:张三。症状：头疼、眼花、失眠。请帮患者生成一份病历。")
    print("=========================")

    desc: str = trans2()
    chat35("你是一个医生，请帮助病人生成一份病历，患者信息: 姓名:李四 年龄：40、性别：女、症状：{0}".format(desc))
    print("=========================")
    print("over")
    #chat("请您继续")

